package io.xpipe.app.util;

import io.xpipe.app.core.AppInstallation;
import io.xpipe.app.core.AppNames;
import io.xpipe.app.core.AppSystemInfo;
import io.xpipe.app.issue.ErrorEventFactory;
import io.xpipe.app.process.CommandBuilder;
import io.xpipe.app.process.LocalShell;
import io.xpipe.app.process.OsFileSystem;
import io.xpipe.app.update.AppDistributionType;
import io.xpipe.core.FilePath;
import io.xpipe.core.OsType;

import java.io.IOException;
import java.nio.file.Files;
import java.nio.file.Path;

public class DesktopShortcuts {

    private static Path createWindowsShortcut(String executable, String args, String name) throws Exception {
        var shortcutPath = AppSystemInfo.ofCurrent().getDesktop().resolve(name + ".lnk");

        var shell = LocalShell.getLocalPowershell();
        if (shell.isEmpty()) {
            Files.createFile(shortcutPath);
            return shortcutPath;
        }

        var icon = AppInstallation.ofCurrent().getLogoPath();
        var content = String.format(
                        """
                                    $TARGET="%s"
                                    $SHORTCUT="%s"
                                    $ws = New-Object -ComObject WScript.Shell
                                    $s = $ws.CreateShortcut("$SHORTCUT")
                                    $S.IconLocation='%s'
                                    $S.WindowStyle=7
                                    $S.TargetPath = "$TARGET"
                                    $S.Arguments = '%s'
                                    $S.Save()
                                    """,
                        executable, shortcutPath, icon, args)
                .replaceAll("\n", ";");
        shell.get().command(content).execute();
        return shortcutPath;
    }

    private static Path getOrCreateIcon() throws IOException {
        if (AppDistributionType.get() != AppDistributionType.APP_IMAGE
                && AppDistributionType.get() != AppDistributionType.NIX) {
            return AppInstallation.ofCurrent().getLogoPath();
        }

        var target = AppSystemInfo.ofCurrent().getUserHome().resolve(".local", "share", "icons", "128x128", "apps");
        var file = target.resolve(AppNames.ofCurrent().getKebapName() + ".png");
        if (Files.exists(file)) {
            return file;
        }

        Files.createDirectories(target);
        Files.copy(AppInstallation.ofCurrent().getLogoPath(), file);
        return file;
    }

    private static Path createLinuxShortcut(String executable, String args, String name) throws Exception {
        // Linux .desktop names are very restrictive
        var fixedName = name.replaceAll("[^\\w _]", "");
        var icon = getOrCreateIcon();
        var content = String.format(
                """
                                    [Desktop Entry]
                                    Type=Application
                                    Name=%s
                                    Comment=Open with XPipe
                                    Exec="%s" %s
                                    Icon=%s
                                    Terminal=false
                                    Categories=Utility;Development;
                                    """,
                fixedName, executable, args, icon);

        var osFile = Path.of("/etc/os-release");
        var ubuntu =
                Files.exists(osFile) && Files.readString(osFile).toLowerCase().contains("ubuntu");
        var file = ubuntu
                ? AppSystemInfo.ofCurrent().getDesktop().resolve(name + ".desktop")
                : AppSystemInfo.ofCurrent().getUserHome().resolve(".local", "share", "applications", name + ".desktop");
        Files.createDirectories(file.getParent());
        Files.writeString(file, content);
        file.toFile().setExecutable(true);

        // Mark shortcuts as trusted on gnome
        LocalShell.getShell()
                .command(CommandBuilder.of()
                        .add("gio", "set")
                        .addFile(file)
                        .addQuoted("metadata::trusted")
                        .add("true"))
                .executeAndCheck();

        return file;
    }

    private static Path createMacOSShortcut(String executable, String args, String name) throws Exception {
        var icon = AppInstallation.ofCurrent().getLogoPath();
        var assets = icon.getParent().resolve("Assets.car");
        var base = AppSystemInfo.ofCurrent().getDesktop().resolve(name + ".app");
        var content = String.format(
                """
                                    #!/usr/bin/env sh
                                    "%s" %s
                                    """,
                executable, args);

        try (var pc = LocalShell.getShell()) {
            pc.getShellDialect().deleteFileOrDirectory(pc, base.toString()).executeAndCheck();
            pc.executeSimpleCommand(pc.getShellDialect().getMkdirsCommand(base + "/Contents/MacOS"));
            pc.executeSimpleCommand(pc.getShellDialect().getMkdirsCommand(base + "/Contents/Resources"));

            var macExec = base + "/Contents/MacOS/" + name;
            pc.view().writeScriptFile(FilePath.of(macExec), content);
            pc.executeSimpleCommand("chmod ugo+x \"" + macExec + "\"");

            pc.view().writeTextFile(FilePath.of(base + "/Contents/PkgInfo"), "APPL????");
            pc.view()
                    .writeTextFile(
                            FilePath.of(base + "/Contents/Info.plist"),
                            """
                                                                                <?xml version="1.0" encoding="UTF-8"?>
                                                                                <!DOCTYPE plist PUBLIC "-//Apple//DTD PLIST 1.0//EN" "http://www.apple.com/DTDs/PropertyList-1.0.dtd">
                                                                                <plist version="1.0">
                                                                                <dict>
                                                                                    <key>CFBundleIconName</key>
                                                                                    <string>xpipe</string>
                                                                                	<key>CFBundleIconFile</key>
                                                                                	<string>xpipe</string>
                                                                                </dict>
                                                                                </plist>
                                                                                """);
            pc.command("cp \"" + icon + "\" \"" + base + "/Contents/Resources/xpipe.icns\"")
                    .execute();
            pc.command("cp \"" + assets + "\" \"" + base + "/Contents/Resources/Assets.car\"")
                    .execute();
        }
        return base;
    }

    public static Path createOpen(String name, String cliArgs, String desktopArgs) throws Exception {
        var compat = OsFileSystem.ofLocal().makeFileSystemCompatible(name);
        if (OsType.ofLocal() == OsType.WINDOWS) {
            var exec = AppInstallation.ofCurrent().getCliExecutablePath().toString();
            return createWindowsShortcut(exec, cliArgs, compat);
        } else if (OsType.ofLocal() == OsType.LINUX) {
            // AppImages are mounted and can't be called normally
            if (AppDistributionType.get() == AppDistributionType.APP_IMAGE) {
                if (desktopArgs == null) {
                    throw ErrorEventFactory.expected(
                            new UnsupportedOperationException("This desktop shortcut operation is not supported in the "
                                    + AppDistributionType.get()
                                            .toTranslatedString()
                                            .getValue() + " distribution"));
                }

                var exec = System.getenv("APPIMAGE");
                return createLinuxShortcut(exec, desktopArgs, compat);
            }

            // Nix store locations change on updates
            // These installations always add the executable to the path
            if (AppDistributionType.get() == AppDistributionType.NIX) {
                return createLinuxShortcut(AppNames.ofCurrent().getKebapName(), cliArgs, compat);
            }

            var exec = AppInstallation.ofCurrent().getCliExecutablePath().toString();
            return createLinuxShortcut(exec, cliArgs, compat);
        } else {
            var exec = AppInstallation.ofCurrent().getCliExecutablePath().toString();
            return createMacOSShortcut(exec, cliArgs, compat);
        }
    }
}
